package com.snake.util;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.ConnectException;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import javax.net.ssl.TrustManager;

import net.sf.json.JSONObject;

import org.apache.http.HttpEntity;
import org.apache.http.HttpResponse;
import org.apache.http.NameValuePair;
import org.apache.http.ParseException;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.client.entity.UrlEncodedFormEntity;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.client.methods.HttpUriRequest;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.apache.http.message.BasicHeader;
import org.apache.http.message.BasicNameValuePair;
import org.apache.http.protocol.HTTP;
import org.apache.http.util.EntityUtils;



public class HttpClientUtil {
	public static String[] post(String url, Map<String, String> params, Map<String, String> headers) {
		CloseableHttpClient httpclient = null;
		String[] body = null;
		try {
			// post 执行体
			httpclient = HttpClients.createDefault();
			HttpPost post = postForm(url, params, headers);
			body = invoke(httpclient, post);
		} catch (Exception e) {
			//logger.error(e);
		} finally {
			if (httpclient != null){
				try {
					httpclient.close();
					httpclient = null;
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		};
		return body;
	}

	public static String[] get(String url, Map<String, String> headers) {
		CloseableHttpClient httpclient = null;
		String[] body = null;
		try {
			httpclient = HttpClients.createDefault();
			HttpGet get = new HttpGet(url);
			// 设置header
			if (headers != null){
				Set<String> keySet = headers.keySet();
				for (String key : keySet) {
					get.setHeader(new BasicHeader(key, headers.get(key)));
				}
			}
			body = invoke(httpclient, get);
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			if (httpclient != null){
				try {
					httpclient.close();
					httpclient = null;
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
		return body;
	}

	private static String[] invoke(CloseableHttpClient httpclient, HttpUriRequest httpost) throws IOException {

		CloseableHttpResponse response = sendRequest(httpclient, httpost);
		String[] body = paseResponse(response);
		response.close();
		return body;
	}

	private static String[] paseResponse(HttpResponse response) {
		HttpEntity entity = response.getEntity();
		String[] responseBody = new String[2];
		try {
			responseBody[0] = String.valueOf(response.getStatusLine().getStatusCode());
			responseBody[1] = EntityUtils.toString(entity);
		} catch (ParseException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		} 
		return responseBody;
	}

	private static CloseableHttpResponse sendRequest(CloseableHttpClient httpclient, HttpUriRequest httpost) {
		CloseableHttpResponse response = null;
		try {
			response = httpclient.execute(httpost);
		} catch (ClientProtocolException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return response;
	}

	@SuppressWarnings("deprecation")
	private static HttpPost postForm(String url, Map<String, String> params, Map<String, String> headers) {

		HttpPost httpost = new HttpPost(url);
		List<NameValuePair> nvps = new ArrayList<NameValuePair>();

		// 设置参数
		if (params != null){
			Set<String> keySet = params.keySet();
			for (String key : keySet) {
				nvps.add(new BasicNameValuePair(key, String.valueOf(params.get(key))));
			}
		}
		
		// 设置header
		if (headers != null){
			Set<String> keySet = headers.keySet();
			for (String key : keySet) {
				httpost.setHeader(key, headers.get(key));
			}
		}

		try {
			httpost.setEntity(new UrlEncodedFormEntity(nvps, HTTP.UTF_8));
		} catch (UnsupportedEncodingException e) {
			e.printStackTrace();
		}

		return httpost;
	}
	/**
	 * 发起https请求并获取结果
	 * 
	 * @param requestUrl
	 *            请求地址
	 * @param requestMethod
	 *            请求方式（GET、POST）
	 * @param outputStr
	 *            提交的数据
	 * @return JSONObject(通过JSONObject.get(key)的方式获取json对象的属性值)
	 */
	public static   JSONObject httpRequest(String requestUrl,
			String requestMethod, String outputStr) {
		JSONObject jsonObject = null;
		StringBuffer buffer = new StringBuffer();
		try {
			// 创建SSLContext对象，并使用我们指定的信任管理器初始化
			TrustManager[] tm = { new MyX509TrustManager() };
			SSLContext sslContext = SSLContext.getInstance("SSL", "SunJSSE");
			sslContext.init(null, tm, new java.security.SecureRandom());
			// 从上述SSLContext对象中得到SSLSocketFactory对象
			SSLSocketFactory ssf = sslContext.getSocketFactory();

			URL url = new URL(requestUrl);
			HttpsURLConnection httpUrlConn = (HttpsURLConnection) url
					.openConnection();
			httpUrlConn.setSSLSocketFactory(ssf);

			httpUrlConn.setDoOutput(true);
			httpUrlConn.setDoInput(true);
			httpUrlConn.setUseCaches(false);
			// 设置请求方式（GET/POST）
			httpUrlConn.setRequestMethod(requestMethod);

			if ("GET".equalsIgnoreCase(requestMethod))
				httpUrlConn.connect();

			// 当有数据需要提交时
			if (null != outputStr) {
				OutputStream outputStream = httpUrlConn.getOutputStream();
				// 注意编码格式，防止中文乱码
				outputStream.write(outputStr.getBytes("UTF-8"));
				outputStream.close();
			}

			// 将返回的输入流转换成字符串
			InputStream inputStream = httpUrlConn.getInputStream();
			InputStreamReader inputStreamReader = new InputStreamReader(
					inputStream, "utf-8");
			BufferedReader bufferedReader = new BufferedReader(
					inputStreamReader);

			String str = null;
			while ((str = bufferedReader.readLine()) != null) {
				buffer.append(str);
			}
			bufferedReader.close();
			inputStreamReader.close();
			// 释放资源
			inputStream.close();
			inputStream = null;
			httpUrlConn.disconnect();
			jsonObject = JSONObject.fromObject(buffer.toString());
		} catch (ConnectException ce) {
			//log.error("Weixin server connection timed out.");
		} catch (Exception e) {
			//log.error("https request error:{}", e);
		}
		return jsonObject;
	}
}
